import type { MenuRecordRaw } from '@nice/types'

import type { ComputedRef } from 'vue'

import { computed, ref, watch } from 'vue'
import { useRoute } from 'vue-router'

import { useAccessStore, useUserStore } from '#/stores/modules'
import { findRootMenuByPath } from '@nice/utils'

import { useNavigation } from './use-navigation'

function useExtraMenu(useRootMenus?: ComputedRef<MenuRecordRaw[]>) {
  const accessStore = useAccessStore()
  const userStore = useUserStore()
  const { navigation, willOpenedByWindow } = useNavigation()

  const menus = computed(() => useRootMenus?.value ?? accessStore.accessMenus)

  /** 记录当前顶级菜单下哪个子菜单最后激活 */
  const defaultSubMap = new Map<string, string>()
  const extraRootMenus = ref<MenuRecordRaw[]>([])
  const route = useRoute()
  const extraMenus = ref<MenuRecordRaw[]>([])
  const sidebarExtraVisible = ref<boolean>(false)
  const extraActiveMenu = ref('')
  const parentLevel = computed(() => 0)

  /**
   * 选择混合菜单事件
   * @param menu
   */
  const handleMixedMenuSelect = async (menu: MenuRecordRaw) => {
    const _extraMenus = menu?.children ?? []
    const hasChildren = _extraMenus.length > 0

    if (!willOpenedByWindow(menu.path)) {
      extraMenus.value = _extraMenus ?? []
      extraActiveMenu.value = menu.parents?.[parentLevel.value] ?? menu.path
      sidebarExtraVisible.value = hasChildren
    }

    if (!hasChildren) {
      await navigation(menu.path)
    } else if (userStore.preferences.sidebar.autoActivateChild) {
      await navigation(
        defaultSubMap.has(menu.path) ? (defaultSubMap.get(menu.path) as string) : menu.path
      )
    }
  }

  /**
   * 选择默认菜单事件
   * @param menu
   * @param rootMenu
   */
  const handleDefaultSelect = async (menu: MenuRecordRaw, rootMenu?: MenuRecordRaw) => {
    extraMenus.value = rootMenu?.children ?? extraRootMenus.value ?? []
    extraActiveMenu.value = menu.parents?.[parentLevel.value] ?? menu.path

    if (userStore.preferences.sidebar.expandOnHover) {
      sidebarExtraVisible.value = extraMenus.value.length > 0
    }
  }

  /**
   * 侧边菜单鼠标移出事件
   */
  const handleSideMouseLeave = () => {
    if (userStore.preferences.sidebar.expandOnHover) {
      return
    }

    const { findMenu, rootMenu, rootMenuPath } = findRootMenuByPath(menus.value, route.path)
    extraActiveMenu.value = rootMenuPath ?? findMenu?.path ?? ''
    extraMenus.value = rootMenu?.children ?? []
  }

  const handleMenuMouseEnter = (menu: MenuRecordRaw) => {
    if (!userStore.preferences.sidebar.expandOnHover) {
      const { findMenu } = findRootMenuByPath(menus.value, menu.path)
      extraMenus.value = findMenu?.children ?? []
      extraActiveMenu.value = menu.parents?.[parentLevel.value] ?? menu.path
      sidebarExtraVisible.value = extraMenus.value.length > 0
    }
  }

  function calcExtraMenus(path: string) {
    const currentPath = route.meta?.activePath || path
    const { findMenu, rootMenu, rootMenuPath } = findRootMenuByPath(
      menus.value,
      currentPath,
      parentLevel.value
    )
    extraRootMenus.value = rootMenu?.children ?? []
    if (rootMenuPath) defaultSubMap.set(rootMenuPath, currentPath)
    extraActiveMenu.value = rootMenuPath ?? findMenu?.path ?? ''
    extraMenus.value = rootMenu?.children ?? []
    if (userStore.preferences.sidebar.expandOnHover) {
      sidebarExtraVisible.value = extraMenus.value.length > 0
    }
  }

  watch(
    () => [route.path],
    ([path]) => {
      calcExtraMenus(path || '')
    },
    { immediate: true }
  )

  return {
    extraActiveMenu,
    extraMenus,
    handleDefaultSelect,
    handleMenuMouseEnter,
    handleMixedMenuSelect,
    handleSideMouseLeave,
    sidebarExtraVisible,
  }
}

export { useExtraMenu }
